package warmachineroster.grapheditor.command;

import org.eclipse.gef.commands.Command;

import warmachineroster.WarcasterRosterElement;
import warmachineroster.WarjackRosterElement;

public class WarjackRosterElementRemoveCommand extends Command {

	private WarjackRosterElement warjack;
	private WarcasterRosterElement warcaster;

	

	public void setWarjack(WarjackRosterElement warjack) {
		this.warjack = warjack;
	}

	public void setWarcaster(WarcasterRosterElement warcaster) {
		this.warcaster = warcaster;
	}

	@Override
	public void execute() {
		warcaster.getBattlegroup().remove(warjack);
	}

	@Override
	public void undo() {
		warcaster.getBattlegroup().add(warjack);
	}
	
}
